package com.rymcu.forest.rag;

import com.rymcu.forest.entity.Article;
import lombok.Builder;
import lombok.Data;

import java.util.List;
import java.util.Map;

/**
 * RAG查询结果模型
 */
@Data
@Builder
public class RAGQueryResult {

    /**
     * 用户原始问题
     */
    private String question;

    /**
     * 生成的SQL语句
     */
    private String sql;

    /**
     * 数据库查询结果
     */
    private List<Map<String, Object>> queryResults;

    /**
     * 相关文章列表
     */
    private List<Article> articles;

    /**
     * AI回答上下文
     */
    private String context;

    /**
     * 错误信息
     */
    private String error;

    /**
     * 是否成功
     */
    public boolean isSuccess() {
        return error == null || error.isEmpty();
    }
}